package com.example.demo1.samplews;

import java.util.ArrayList;
import java.util.List;

public class AiAgentChatRoom {

    List<AiAgentWs> aiAgents = new ArrayList<>();

    public void accept(AiAgentWs aiAgent) {
        this.aiAgents.add(aiAgent);
        System.out.println("【" + aiAgent.getModel() + "】 加入对话");
    }

    public static void main(String[] args) {
        chatOfFreeAgents();
    }

    public static void chatOfFreeAgents() {
        AiAgentHttp llama3p1_8b = new AiAgentHttp("http://localhost:11434/api/chat", "llama3.1:8b", 10, "可以聊聊吗", "关于当前社会的方方面面随机展开探讨一些问题，任何方面都可以。回答尽量简短、口语化，不要超过20字。");
        AiAgentHttp qwen_0p5b = new AiAgentHttp("http://localhost:11434/api/chat", " qwen:0.5b ", 10, "我想聊聊", "关于当前社会的方方面面随机展开探讨一些问题，任何方面都可以。回答尽量简短、口语化，不要超过20字。");
        llama3p1_8b.setPartner(qwen_0p5b);
        qwen_0p5b.setPartner(llama3p1_8b);
        llama3p1_8b.start();
    }

}
